from django_redis import get_redis_connection
from rest_framework import serializers

from oauth.models import OAuthQQUser
from oauth.utils import OAuthQQ
from users.models import User


class QQAuthUserSerializer(serializers.Serializer):
     """
     QQ登录创建⽤户序列化器
     """
     access_token = serializers.CharField(label='操作凭证')
     mobile = serializers.RegexField(label='⼿机号', regex=r'^1[3-9]\d{9}$')
     password = serializers.CharField(label='密码', max_length=20, min_length=8)
     sms_code = serializers.CharField(label='短信验证码')
     def validate(self, data):
     # 检验access_token
         access_token = data['access_token']
         # 获取身份凭证
         openid = OAuthQQ.check_save_user_token(access_token)
         if not openid:
            raise serializers.ValidationError('⽆效的access_token')
         # 将openid放在校验字典中，后⾯会使⽤
         data['openid'] = openid
         # 检验短信验证码
         mobile = data['mobile']
         sms_code = data['sms_code']
         redis_conn = get_redis_connection('verify_codes')
         real_sms_code = redis_conn.get('sms_%s' % mobile)
         if real_sms_code.decode() != sms_code:
            raise serializers.ValidationError('短信验证码错误')
         # 如果⽤户存在，检查⽤户密码
         try:
            user = User.objects.get(mobile=mobile)
         except User.DoesNotExist:
            pass
         else:
             password = data['password']
             if not user.check_password(password):
                raise serializers.ValidationError('密码错误')
                # 将认证后的user放进校验字典中，后续会使⽤
             data['user'] = user
         return data
     def create(self, validated_data):
         # 获取校验的⽤户
         user = validated_data.get('user')
         if not user:
         # ⽤户不存在,新建⽤户
             user = User.objects.create_user(
                 username=validated_data['mobile'],
                 password=validated_data['password'],
                 mobile=validated_data['mobile'],
             )
         # 将⽤户绑定openid
         OAuthQQUser.objects.create(
             openid=validated_data['openid'],
             user=user
         )
         # 返回⽤户数据
         return user